from haystack.utils.highlighting import Highlighter

class CustomHighlighter(Highlighter):
    # 自定义高亮类，继承自haystack的Highlighter
    def __init__(self, query, **kwargs):
        # 初始化时设置高亮的参数
        self.query = query
        self.max_length = kwargs.get('max_length', 200)  # 最大高亮长度
        self.css_class = kwargs.get('css_class', 'highlighted')  # 高亮CSS类名
        self.html_tag = kwargs.get('html_tag', 'span')  # 高亮HTML标签
        self.max_words = kwargs.get('max_words', None)  # 最大词数
        self.strip = kwargs.get('strip', True)  # 是否去除空白